mttkrp_shapes = [
    (128, 256, 512, 64),
    (128, 256, 256, 128),
    (256, 128, 256, 128),
    (256, 128, 128, 256),
    (512, 64, 128, 256),
    (512, 64, 64, 512),
    (1024, 32, 32, 1024)
]